package com.wx.leetcode;

/**
 * @desc: 移除元素
 * 给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。
 * 不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
 * 元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
 * 给定 nums = [0,1,2,2,3,0,4,2], val = 2,
 * 函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
 * @author: wx
 * @time: 2020/4/14 12:47 PM
 */
public class RemoveElementTopic {

    public static int removeElement(int[] nums, int val) {

        if (nums == null || nums.length == 0) {
            return 0;
        }

        if (nums.length == 1) {
            return nums[0] == val ? 0 : 1;
        }


        int endIndex = nums.length;

        for (int i = 0; i < endIndex ;) {
             if (nums[i] == val){
                 nums[i] = nums[--endIndex];
             } else {
                 i++;
             }

        }
        return endIndex;
    }

    public static void main(String args[]) {
        int[] nums = new int[]{3,3};
        int len = removeElement(nums, 3);
        System.out.println("长度:" + len);
        for (int i = 0; i < len; i++) {
            System.out.print(nums[i]);
        }

    }


}
